Adding the wireframe plotting part... it's still not quite working properly