Quick refactoring.
Different enemy type support.