Added some logic checking for correct songs.