Improved generator, added cases
Added problems
Added problems