updated README
simplify code