Fix functions 'answer', 'square' and 'average' to work as intended