Refactor README and add Pytest as test dependency