Clean code, add enemy class, remove unused code