updated paper and code