big fixes + tweeks