Will add more unit tests after UI implementation begins