Improve main example