added some code to the comments that will work later