add styling that works