MEDIUM / Working on unit tests