added unittest for finals prep