n_var -> n_features

Also, the shape description should be on line above.