Add some code for security and manage roles