* Add Sagemaker Model Storage * Fix checkstyle * Sagemaker unittest * Small refactor to be less verbose