Trying to implement tests