further improve code