Add prefabs for powerups
Refactor states