small fix & changed usage a little