main() class updated