Style fixes

Thanks!