Fixed bug and Changed documentation