Fixed missing return and public keyword