Clean up graph drawing