small fix + addition