Unit Tests

very basic Unit Tests to be updated along with code