Adding restart and play/pause