ch04-02 code clean up and comment