Adding code, updating readme