Added a name argument

Hopefully I didn't break anything (yolo)