Code update, add arrays