Update media model

Update media model