Added polish, fixed mistake