Add small changes in albums_controller