Our research paper (and code) for COMP 4420 at University of Manitoba