Add some nice styling

feel free to merge ;)