refactor main, prepare_data: moved code into setup_cluster_assignment
refactoring *: ups, functions and methods are smallCamelCased, not underscored