Adjust README towards the actual model