Refactored logic for play button (to be synced with the state).