Turning this into an easy python module with some easy initialisation commands