Project works, just needs correct output