Corrected and written better, still not right answer