add to main to make it run