Working, need to implement output