Save current code for unit test.
Will need to revisit later.