Need advice on writing a program!