Filled out some utils, fixed bugs, added inference test