adding back button, minor styling and also removal of {album_id
code