small mods - final v2