Just added return statement to return void instead of last line