change main for compiling together with dijstra