fixed chpt 4 issues