Does a bit of refactoring

Adds missing comma