Add checks in controllers

what about success message here?