fixed imports
improved documentation
fixed bugs