todo + fixes