Get types to work out, needs work