Increase similarity to original code