Made code work
Made code work