fixed little error
modified a few things