improve ng main controller