Forked the library to have more control