Added necessary return statement