More css classes for easier styling

I'm not sure whether I'm using the module correctly, but when I request a photoset, either as a node or using 'views' / field, it just returns one lump of tags. No titles are displayed.

So I've done a temporary hack to the main module file:

function theme_flickr_photoset($ps, $owner, $size, $attribs = NULL) {
  if (module_exists('flickr_sets')) {
	$output .= "<div class='flickr-photoset'>\n";
	$photos = flickr_set_load($ps['id']);
	foreach ((array) $photos['photoset']['photo'] as $photo) {
	  //insert owner into $photo because theme_flickr_photo needs it
	  $photo['owner'] = $owner;
	  $output .= "<div class='flickr-photoset-frame'>\n";
	  $output .= "<div class='flickr-photoset-photo'>\n";
	  $output .= theme('flickr_photo', $photo, $size) . "\n";
	  $output .= "</div>\n";
	  $output .= "<div class='flickr-photoset-title'>\n";
	  $output .= $photo['title'];
	  $output .= "</div>\n";
	  $output .= "</div>\n";
	}
    $output .= '</div>';
	return $output;
  } else {
    $img = flickr_img($ps, $size, $attribs);    
    $output = theme('pager', NULL, variable_get('flickr_photos_per_page', 20));
    $photo_url = flickr_photoset_page_url($owner, $ps['id']);    
    $output .= "<div class='flickr-photoset'>\n";
    $title = is_array($ps['title']) ? $ps['title']['_content'] : $ps['title'];   
    return l($img, $photo_url, array('attributes' => array('title' => $title), 'absolute' => TRUE, 'html' => TRUE));
  }
}

Which seems to work. Is there a better way of achieving the end result?

See http://www.gear7.co.uk/gallery for an example.