Add missing return statement and fix headers capitalization