Working version after Lecture 3
Added 4 more buttons