added code and update readme