Additional work for PR #18

Completed the work for PR #18. Apologies for the number of commits here, I didn't want to rebase and mess everything up.
The linter doesn't warn on this but I feel that =s should be lined up just like =>s