Add initial rough code