Made tracks and albums buttons code more generic for the different sizes