Fix frame centering issues with checkout and switch subcommands

Signed-off-by: Jacob Stopak <jacob@initialcommit.io>

Author: initialcommit-io

git_sim/checkout.py

@@ -72,6 +72,8 @@ def construct(self):
         # using -b flag, create new branch label and exit
         if self.b:
             self.parse_commits(head_commit)
+            self.recenter_frame()
+            self.scale_frame()
             self.draw_ref(head_commit, self.topref, text=self.branch, color=m.GREEN)
         else:
             branch_commit = self.get_commit(self.branch)

git_sim/switch.py

@@ -72,6 +72,8 @@ def construct(self):
         # using -c flag, create new branch label and exit
         if self.c:
             self.parse_commits(head_commit)
+            self.recenter_frame()
+            self.scale_frame()
             self.draw_ref(head_commit, self.topref, text=self.branch, color=m.GREEN)
         else:
             branch_commit = self.get_commit(self.branch)