Resolved issue #1

Hi, I have patched your code to support not rounded layers as well. Please accept the pull request if you like the solution.