TODO: fix todo's