Beauty up the code

Removed useless class reference.