Style updates

Thanks @tekin