Cleaned up code indentation