fixing minor stuff