added prob 1 and 2