small bug fix and added initial problems