Cleans up some code samples in docs.