public View onCreateView(LayoutInflater inflater, ViewGroup container,
                             Bundle savedInstanceState) {
        // Inflate the layout for this fragment
        View view = inflater.inflate(R.layout.fragment_movie, container, false);

        //Initialize
        ButterKnife.bind(this, view);
        context = getActivity();
        celebrityActivity = (CelebrityActivity) getActivity();
        isCollapsing = celebrityActivity.getCollapsing();

        //Get counts
        actorCount = celebrityActivity.movieActorCount();
        directorCount = celebrityActivity.movieDirectorCount();
        writerCount = celebrityActivity.movieWriterCount();

        //Set actor tab
        if (actorCount > 0) {
            celebrityMovieFrag_tabLayout.addTab(celebrityMovieFrag_tabLayout.newTabPlus());
            badgedTabCustomView = celebrityMovieFrag_tabLayout.getTabCustomViewAt(0);
            if (badgedTabCustomView != null) {
                badgedTabCustomView.setTabText("Actor");
                badgedTabCustomView.setTabCount(actorCount);
            }
        }
        //Set director tab
        if (directorCount > 0 && actorCount == 0) {
            celebrityMovieFrag_tabLayout.addTab(celebrityMovieFrag_tabLayout.newTabPlus());
            badgedTabCustomView = celebrityMovieFrag_tabLayout.getTabCustomViewAt(0);
            if (badgedTabCustomView != null) {
                badgedTabCustomView.setTabText("Director");
                badgedTabCustomView.setTabCount(directorCount);
            }
        } else if (directorCount > 0) {
            celebrityMovieFrag_tabLayout.addTab(celebrityMovieFrag_tabLayout.newTabPlus());
            badgedTabCustomView = celebrityMovieFrag_tabLayout.getTabCustomViewAt(1);
            if (badgedTabCustomView != null) {
                badgedTabCustomView.setTabText("Director");
                badgedTabCustomView.setTabCount(directorCount);
            }
        }
        //Set writer tab
        if (writerCount > 0 && directorCount == 0 && actorCount == 0) {
            celebrityMovieFrag_tabLayout.addTab(celebrityMovieFrag_tabLayout.newTabPlus());
            badgedTabCustomView = celebrityMovieFrag_tabLayout.getTabCustomViewAt(0);
            if (badgedTabCustomView != null) {
                badgedTabCustomView.setTabText("Writer");
                badgedTabCustomView.setTabCount(writerCount);
            }
        } else if (writerCount > 0 && directorCount > 0 && actorCount == 0) {
            celebrityMovieFrag_tabLayout.addTab(celebrityMovieFrag_tabLayout.newTabPlus());
            badgedTabCustomView = celebrityMovieFrag_tabLayout.getTabCustomViewAt(1);
            if (badgedTabCustomView != null) {
                badgedTabCustomView.setTabText("Writer");
                badgedTabCustomView.setTabCount(writerCount);
            }
        } else if (writerCount > 0) {
            celebrityMovieFrag_tabLayout.addTab(celebrityMovieFrag_tabLayout.newTabPlus());
            badgedTabCustomView = celebrityMovieFrag_tabLayout.getTabCustomViewAt(2);
            if (badgedTabCustomView != null) {
                badgedTabCustomView.setTabText("Writer");
                badgedTabCustomView.setTabCount(writerCount);
            }
        }
        //Tabs gravity
        celebrityMovieFrag_tabLayout.setTabGravity(TabLayout.GRAVITY_FILL);
        //Creating our pager adapter
        Pager adapter = new Pager(getActivity().getSupportFragmentManager(), celebrityMovieFrag_tabLayout.getTabCount());
        //Adding adapter to pager
        celebrityMovieFrag_viewPager.setAdapter(adapter);
        //Disable Swipe viewPager
        celebrityMovieFrag_viewPager.disableScroll(true);
        //Adding onTabSelectedListener to swipe views
        celebrityMovieFrag_tabLayout.setOnTabSelectedListener(new TabLayout.OnTabSelectedListener() {
            @Override
            public void onTabSelected(TabLayout.Tab tab) {
                celebrityMovieFrag_viewPager.setCurrentItem(tab.getPosition());
            }

            @Override
            public void onTabUnselected(TabLayout.Tab tab) {

            }

            @Override
            public void onTabReselected(TabLayout.Tab tab) {

            }
        });

        return view;
    }

    public class Pager extends FragmentStatePagerAdapter {

        //integer to count number of tabs
        int tabCount;

        //Constructor to the class
        public Pager(FragmentManager fm, int tabCount) {
            super(fm);
            //Initializing tab count
            this.tabCount = tabCount;
        }

        //Overriding method getItem
        @Override
    public Fragment getItem(int position) {
        //Returning the current tabs
        switch (position) {
            case 0:
                SampleFragment tab1 = new SampleFragment();
                return tab1;
            case 1:
                SampleFragment tab2 = new SampleFragment();
                return tab2;
            case 2:
                SampleFragment tab3 = new SampleFragment();
                return tab3;
            default:
                return null;
        }
    }

        //Overriden method getCount to get the number of tabs
        @Override
        public int getCount() {
            return tabCount;
        }
    }
}