fixing up some code