Hmmm I am honestly not sure how to change the current code into that