Fixed failed test case. Added more tests. Fixed docs build