Improved play/pause button behaviour