Code is working now!