diff --git a/docs/api.rst b/docs/api.rst
index 13457865..38b82fa5 100644
--- a/docs/api.rst
+++ b/docs/api.rst
@@ -62,33 +62,25 @@ This returns the matrix of log posterior probabilities for numerical stability.
 API Reference
 -------------
 
+Distributions
+=============
+
 .. automodule:: pomegranate.distributions
    :members: Bernoulli, Categorical, ConditionalCategorical, JointCategorical, DiracDelta, Exponential, Gamma, Normal, Poisson, StudentT, Uniform, ZeroInflated
 
-.. automodule:: pomegranate.bayes_classifier.BayesClassifier
-	:members:
-	:inherited-members:
+Models
+======
+
+.. autoclass:: pomegranate.bayes_classifier.BayesClassifier
 
-.. automodule:: pomegranate.gmm.GeneralMixtureModel
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.gmm.GeneralMixtureModel
 
-.. automodule:: pomegranate.hmm.DenseHMM
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.hmm.DenseHMM
 
-.. automodule:: pomegranate.hmm.SparseHMM
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.hmm.SparseHMM
 
-.. automodule:: pomegranate.markov_chain.MarkovChain
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.markov_chain.MarkovChain
 
-.. automodule:: pomegranate.bayesian_network.BayesianNetwork
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.bayesian_network.BayesianNetwork
 
-.. automodule:: pomegranate.factor_graph.FactorGraph
-	:members:
-	:inherited-members:
\ No newline at end of file
+.. autoclass:: pomegranate.factor_graph.FactorGraph
\ No newline at end of file