Removed unnecessary test code and corrected doc style