Added ReLu support. Improved test cases