Adding code and updating readme