Format code

Signed-off-by: Robbie Cao <robbie.cao@gmail.com>