Beamline setup - Improved the documentation

Created by: marcus-oscarsson

Hey,

I improved the documentation of beamline_setup.py, also made some minor updates to the API. Also fixed some linting errors.

Thanks !