It works! So cool. Time to make the code look nice