added code
adding test section and enable it in travis