Working on uni t tests