Added Python interface that may be useful for plotting later