code challenge - improvments