move code to src dir
update comment