todo fixes, styling