UI changes
comments and documentation
Plot can be saved to png