Refactore controllers