Returning code to make it work