Mod update 1.0.6

Small fixes