Minor improvement on class inheritance walkthrough