Custom model box redesign for the albums plug in