Assignment SWARM, Python.
TODO: C++ code