Clean up files and improve example