Using inheritance a little better