Adding functionality for 2 player