fixed module5 code