Add more comments and documentation
Write tests
Create rakefile