Add tests for subclassing behavior