adding return statement to make it work :)