i've been working on the final class