two small fixes
add 2 important classes
fixing error and updating documentation