changes to get the code to work