add stuff i forgot in the last commit