Todo updates & tweaks
TODO's