I liked the mechanics. They forced you to change your behavior according to your enemy's color, even though they behaved almost identically; it's also a clever way to reuse the AI. I think the concept it's worth exploring a bit further, maybe through different level layouts to see where it can go.