Finish writing JUnits