Forgot some todo:s