Add method to get type of media item