Remove unused functions and add more test coverage

why do we have this line here? Can we assign the value in __init__()?