Just some visual updates for a nicer look