Fixing sample problem set