Fix outdated main method calls

Had forgot to update this file and um not sure where it went but
it's rewritten so the program runs now! Doh...