move into python and create a new one for java