Add sample main function