Small updates on docs + style