Add checks in controllers

try..catch?