Doc generator class fixed