Task/add unit tests

#### Reviewers
@jframos @flopezag @jesuspg 

#### Description
ADD unit tests

#### Test
tested locally 
this has historically been like this, could be refactored. Ok