Add description and tidy code