'improved' the code