some changes to create a propper python package