We need to add test cases for the code so that travis can make sure it works for us.