Make code more ruby. Start Connection class