changed functions params and added enemy class