5 test cases implememnted + stress tests + some refactoring
generator fix and loop