Updated to include paper that the program is based on