Fixed things i forgot to