make changes to make it all work