adding code and updating readme