Worked and now one function more to test