Fixing minor mistakes