Updated styles to match prototype