Assert Statements work added