make changes into view controller h and m add test method