Added plotting, looks horrible right now