Minor tweaks to program

Fixed a couple minor issues