fixed the code because it broke when I recopied it