Add pause functionality