Slight refactor of biopython visualisation notebook