add main() code