Aded all tests to the Demo project and updated the header comments