Fixed class name
Code formatting