Restart Policy for Containers Implemented fixes #302

This is my attempt to fix #302 let me know if any modifications are needed, willing to comply to the repo standards and contribute.