fixing a few mistakes