Cleanup and complete the code example on homepage