fixed two small problems