Add ability to pause