Handle more than one iteration

Fix kwargs