Simplify code

Updated README accordingly.