Cleaned up the final example