Works, but is a bit shit

I don't think the connection should be a class-level thing