Small update to correct issues

Corrected some minor issues