How exactly would I implement that in my code above?