changed return stmt