fixed my fuckup. Reverted a change to main method