Updated interface and moved number of rounds out of the context.