Updated outdated class names.

Updated readme to match current implementation.