Added tests for units