Fix example
Cleanup