fixes + TODO