adding stuff i forgot to add on last commit :)