Move test and update code comment
2 files changed