More styling fixes, updated TODO