refactor input layer a bit more. closer to testable