Fix bug (change method to methods in __init__)