Updated equality exercise to use only returns