Updated main for testing on nodes