Adding back in actual plotting code.