updated problem 4 bs :) :)