Fixes #10 . Nicer code format