Breaking out class & function into python modules