add some styling
edit controller