minor changes
added plot functions and starting to test