Remove example code from main