Updated class name

Signed-off-by: Michael Reiner <mreiner77@gmail.com>