Fixed some minor issues in the code.