Aded main() function