Update aws code