Updated main() a bit