fixed tests and cleaned up docs