added code and changed README