Adding the first round of unit/feature tests