Fixed indentation
Cleaned up code.