fixed up it a bit

added a couple stuff