diff --git a/src/framework/mod.rs b/src/framework/mod.rs
index c881120..912734e 100644
--- a/src/framework/mod.rs
+++ b/src/framework/mod.rs
@@ -14,6 +14,7 @@
 //! 
 //! ```
 //! use crate::framework::bot::Bot;
+//! use crate::state::State;
 //! 
 //! // Your app's state.
 //! let state = State::new();
@@ -100,7 +101,10 @@
 //! arguments. The following is a full example of a command with all these features:
 //! 
 //! ```
-//! use crate::framework::commands::{Command};
+//! use crate::framework::commands::{
+//!     Command, CommandError, context::CommandContext, arguments::NeedyGreedyString
+//! };
+//! use crate::state::State;
 //! 
 //! pub struct HelloCommand;
 //!