fixed the entire code

ye