fixing bug and getting ready for screngrabs