Fixed unittest. Fixed some documentation