Added basic plotting functionality