added basic plot functionality