Problem 2

Fixing variables