moved logic to stage and added enemy