Trying to implement tests. Need to call