Cleaned up code, added back the enemy code