Refactor to support input and output layers