Added weight methods to test driver, encountered a problem with Floating exception.