Move initialisation to special function

This adds a new method `setup_initialisation` which can be called to run
the initialisation step. Splitting it out from the main functionality
should make it easier for new users.