Added additional code..
Added plot
Updated test program