Fixed a couple small things