Python Pass Variable To Class