made a few changes to get it compiling