Identify and resolve performance bottlenecks

[AntonPetrov]

• construct a test set (could use tests.py as a starting point)
• perform timing experiments to see how long each step takes as a way to identify the bottlenecks and their severity

@nawrockie