Fixed a couple little things