Chenyi's temporary test code. We will get rid of this eventually.