Update style per review comments