Tests are written correctly now