Added album model and tests