Worked out importing classes from modules properly.