Made a small fix, program was broken