Cleaned up indentation in code