Implemented PropNet but a little buggy...