Added tests and refactored tests again for linear design.