Re-added function to the play button