add controller and fixed category display