Small refactoring, add enemy class