Code update + test update