Changed so element_amount isn't passed in the __init__ anymore, rather, you should use generate