refactor code
Add helmet