added function to shit