Change the source under sbt

1. add simple tests