Updated main.py: added enemy class and room