Added missing include directive and return statement