Did a refactor of wraith and updated example accordingly. A few minor bugs to be fixed