updating the code for stupid mistake