Random initialization + setting answers for last layer