Prettify code
Refactor
Merge branch 'feature/refactor'