slight improvement in code