Add code and changed readme