Add easy difficulty, add some minor changes to architecture