Adding ability to specify file and trying to fix algo