small mods

mods