Small changes in album module