Added play/pause button, cleaned up code a bit