Minor update on docs