Worked on code

Signed-off-by: shrimpboyho <shrimpboyhoftw@gmail.com>