Locate derived types in assemblies

Shamelessly nicked from http://www.codemeit.com/code-collection/c-find-all-derived-types-from-assembly.html[^] but extended and tidied, this simple pair of extension methods locate all the derived types of a type in a given assembly. Also, I added the ability to locate a type by name, since sometimes you can't find the namespace, or don't care.

public static class LocateStuffInAssembliesExtension
{
    public static Type FindTypeFromAssembly(this Assembly assembly,
        string name)
    {
        var types = assembly.GetTypes();
        return types.SingleOrDefault(t => t.Name == name);
    }

    public static IEnumerable<Type> FindDerivedTypesFromAssembly(this Assembly assembly,
        Type baseType,
        bool classOnly)
    {
        var types = assembly.GetTypes();
        foreach (var type in types.Where(type => !classOnly || type.IsClass))
        {
            if (baseType.IsInterface)
            {
                var it = type.GetInterface(baseType.FullName);

                if (it != null)
                {
                    yield return type;
                }
            }
            else if (type.IsSubclassOf(baseType))
            {
                yield return type;
            }
        }
    }
}