Fix indentation issues with end
Merge pull request #44 from LnL7/fix-indent

Fix indentation issues with end