refactored project 2 to have Albums class