add function return and fix styles