Minor formatting, fix class name