Fix #349 Increase code climate

# What's Changed
* Though probably not 100% yet, this just combined duplicate code in create_stage and update_stage mutations.

# What's next
* Going to tackle some more low hanging fruit - #341 I think.