fixed some minor issues in code