Added missing return at end of file