Updated setup file and released version 2 to pip