Just fixed up a couple little things