Included a main
Added the example from the description of the assignment
Added __init__