Add first draft code