Fixed a small error
Formatted code