Add ui callback to play button