code challenge improvemnts