Made some small fixes

Yay me!