fixing a few small things