Fixed couple small things