At the moment we don't really have proper tests so it would be nice to have pytest working and full unit tests implemented.