Corrected bug and changed the installation method to pip