Changed around press function a little