added more styling. kinda working branch