v1.4.1, improve code