Fixed some todo:s