Cleaned up clustering notebook